transformers==4.45.2
torch==2.4.0
torchaudio==2.4.0
accelerate==1.5.2
datasets==3.0.2
deepspeed==0.15.0
fire==0.7.0
flash_attn==2.6.1
numpy==1.26.4
vllm==0.6.3.post1
wandb==0.18.3
xformers==0.0.27.post2